package com.atguigu.realtime.app.dws;

import com.atguigu.realtime.app.BaseSqlApp;
import com.atguigu.realtime.bean.ProvinceStats;
import com.atguigu.realtime.util.FlinkSinkUtil;
import org.apache.flink.table.api.Table;
import org.apache.flink.table.api.bridge.java.StreamTableEnvironment;

import static com.atguigu.realtime.commont.Constant.TOPIC_DWM_ORDER_WIDE;

/**
 * @Author lizhenchao@atguigu.cn
 * @Date 2021/12/1 14:12
 */
public class DwsProvinceApp extends BaseSqlApp {
    public static void main(String[] args) {
        new DwsProvinceApp().init(4003, 1, "DwsProvinceApp");
    }
    
    @Override
    protected void run(StreamTableEnvironment tenv) {
        // 1. 建立动态表, 与topic进行关联 (dwm_order_wide)
        tenv.executeSql("create table order_wide(" +
                            "   province_id bigint, " +
                            "   province_name string, " +
                            "   province_area_code string, " +
                            "   province_iso_code string, " +
                            "   province_3166_2_code string, " +
                            "   split_total_amount decimal(20, 2), " +
                            "   order_id bigint, " +
                            "   create_time string, " +
                            "   et as to_timestamp(create_time), " +
                            "   watermark for et as et - interval '3' second " +
                            ")with(" +
                            "   'connector' = 'kafka'," +
                            "   'properties.bootstrap.servers' = 'hadoop162:9092,hadoop163:9092, hadoop164:9092'," +
                            "   'properties.group.id' = 'DwsProvinceApp'," +
                            "   'topic' = '" + TOPIC_DWM_ORDER_WIDE + "'," +
                            "   'scan.startup.mode' = 'latest-offset', " +
                            "   'format' = 'json' " +
                            ")");
        
        // 2. 开窗聚合:  滚东窗口
        Table result = tenv.sqlQuery("select " +
                                        " date_format(tumble_start(et, interval '5' second), 'yyyy-MM-dd HH:mm:ss') stt, " +
                                        " date_format(tumble_end(et, interval '5' second), 'yyyy-MM-dd HH:mm:ss') edt, " +
                                        " province_id, " +
                                        " province_name, " +
                                        " province_area_code area_code, " +
                                        " province_iso_code iso_code, " +
                                        " province_3166_2_code iso_3166_2, " +
                                        " sum(split_total_amount) order_amount, " +
                                        " count(distinct(order_id)) order_count, " +
                                        " unix_timestamp() * 1000 ts " +
                                        "from order_wide " +
                                        "group by " +
                                        "   province_id, " +
                                        "   province_name, " +
                                        "   province_area_code, " +
                                        "   province_iso_code, " +
                                        "   province_3166_2_code, " +
                                        "   tumble(et, interval '5' second)");
    
        // 3. 写入到ClickHouse中  (建立一个动态表与ClickHouse关联)
        // 把结构转成流, 再写入
        tenv
            .toRetractStream(result, ProvinceStats.class)
            .filter(t -> t.f0)
            .map(t -> t.f1)
            .addSink(FlinkSinkUtil.getClickHouseSink(
                "gmall2021", "province_stats_2021", ProvinceStats.class
            ));
            
        
    }
}
